Align types in type annotation Python